package com.heimachengxuyuan.transaction;

import org.apache.commons.lang3.StringUtils;
import org.apache.rocketmq.client.producer.*;
import org.apache.rocketmq.common.message.Message;
import org.apache.rocketmq.common.message.MessageExt;

import java.util.concurrent.TimeUnit;

/**
 * 发送事务消息
 */
public class Producer {

    public static void main(String[] args) throws Exception {
        //1.创建消息生产者producer，并制定生产者组名
        TransactionMQProducer producer = new TransactionMQProducer("group5");
        //2.指定Nameserver地址
        producer.setNamesrvAddr("192.168.25.135:9876;192.168.25.138:9876");

        //添加事务监听器（这时候和其他的就有区别了）
        //这就是执行本地事务的入口
        producer.setTransactionListener(new TransactionListener() {
            /**
             * 在该方法中执行本地事务。这个小demo的效果就是2个消息提交了，1个消息回滚了（mq就把消息给删了）
             * @param msg
             * @param arg
             * @return
             */
            @Override
            public LocalTransactionState executeLocalTransaction(Message msg, Object arg) {
                //如果是taga的消息，我就让它提交。
                if (StringUtils.equals("TAGA", msg.getTags())) {
                    return LocalTransactionState.COMMIT_MESSAGE;
                    //如果消息的tag是tagb，我就让它回滚。
                } else if (StringUtils.equals("TAGB", msg.getTags())) {
                    return LocalTransactionState.ROLLBACK_MESSAGE;
                } else if (StringUtils.equals("TAGC", msg.getTags())) {
                    //如果消息的tag是tagc，我就不处理。那mq没有得到你对消息的一个明确处理，那就会回查。
                    return LocalTransactionState.UNKNOW;
                }
                return LocalTransactionState.UNKNOW;
            }

            /**
             * 该方法时MQ进行消息事务状态回查，tagc那个消息就会到这来。   mq一直没收到commit或者rollback，那就回查
             * @param msg
             * @return
             */
            @Override
            public LocalTransactionState checkLocalTransaction(MessageExt msg) {
                System.out.println("消息的Tag:" + msg.getTags());
                return LocalTransactionState.COMMIT_MESSAGE;
            }
        });

        //3.启动producer
        producer.start();

        String[] tags = {"TAGA", "TAGB", "TAGC"};

        for (int i = 0; i < 3; i++) {
            //4.创建消息对象，指定主题Topic、Tag和消息体
            /**
             * 参数一：消息主题Topic
             * 参数二：消息Tag，这里用三个不同tag的消息去分别演示。发送成功本地提交，本地事务执行失败进行回滚，以及不做处理让MQ对消息进行回查。
             * 参数三：消息内容
             */
            Message msg = new Message("TransactionTopic", tags[i], ("Hello World" + i).getBytes());
            //5.发送消息， 这一步就是图中的send half msg
            /**
             * 第二个参数是：object。
             * 就是在事务控制的时候，我们可以将事务控制应用到某一个消息上面，也可以应用到整个producer上面。应用到producer上之后，它上面发送到消息都可以进行一个事务控制。
             */
            SendResult result = producer.sendMessageInTransaction(msg, null);
            //发送状态
            SendStatus status = result.getSendStatus();

            System.out.println("发送结果:" + result);

            //线程睡1秒
            TimeUnit.SECONDS.sleep(2);
        }

        //6.关闭生产者producer、   由于后面mq要回查消息，所以生产者先别停了，不然就回查不到了。
        //producer.shutdown();
    }
}
